import onnxruntime as ort
import numpy as np

# 1️⃣ 加载 ONNX 模型
session = ort.InferenceSession("simple_net.onnx")

# 2️⃣ 查看模型输入输出信息
print("模型输入名称：", session.get_inputs()[0].name)
print("模型输出名称：", session.get_outputs()[0].name)

# 3️⃣ 构造一个随机输入（1行4列）
x = np.random.randn(1, 4).astype(np.float32)
print("输入数据：", x)

# 4️⃣ 执行推理
outputs = session.run(None, {"input": x})

# 5️⃣ 打印结果
print("模型输出结果：", outputs)
